#ifndef SNCascadeBoostTree_h__
#define SNCascadeBoostTree_h__

#include "opencv2\ml\ml.hpp"

class SNCascadeBoostTree : public CvBoostTree
{
public:
	SNCascadeBoostTree();
	~SNCascadeBoostTree();

public:
	//virtual CvDTreeNode* predict(int sampleIdx) const;
	void write(cv::FileStorage &fs, const cv::Mat& featureMap);
	void read(const cv::FileNode &node, CvBoost* _ensemble, CvDTreeTrainData* _data);
	void markFeaturesInMap(cv::Mat& featureMap);
protected:
	//virtual void split_node_data(CvDTreeNode* n);
};

#endif // SNCascadeBoostTree_h__
